package com.monotoneStack;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Stack;

/**
 * 下一个更大元素
 * 输入：nums1 = [4,1,2], nums2 = [1,3,4,2].
 * 输出：[-1,3,-1]
 * 解释：nums1 中每个值的下一个更大元素如下所述：
 * - 4 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
 * - 1 ，用加粗斜体标识，nums2 = [1,3,4,2]。下一个更大元素是 3 。
 * - 2 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1
 */
public class NextGreaterElementI0121_496 {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        //单调栈 HashMap
        Stack<Integer> stack = new Stack<>();
        int[] ans = new int[nums1.length];
        Arrays.fill(ans,-1);

        HashMap<Integer,Integer> map = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            map.put(nums1[i],i);
        }

        stack.push(0);
        for (int i = 1; i < nums2.length; i++) {
            if(nums2[i] <= nums2[stack.peek()]){
                stack.push(i);
            }else {
                while (!stack.isEmpty() && nums2[i] > nums2[stack.peek()]){
                    if(map.containsKey(nums2[stack.peek()]) ){
                        ans[map.get(nums2[stack.peek()])] = nums2[i];
                    }
                    stack.pop();
                }
                stack.push(i);
            }
        }
        return ans;
    }
}
